home *** CD-ROM | disk | FTP | other *** search
- # Source Generated with Decompyle++
- # File: in.pyo (Python 2.4)
-
- import os
- import unittest
- import warnings
- from test import test_support
- warnings.filterwarnings('ignore', 'tempnam', RuntimeWarning, __name__)
- warnings.filterwarnings('ignore', 'tmpnam', RuntimeWarning, __name__)
-
- class TemporaryFileTests(unittest.TestCase):
-
- def setUp(self):
- self.files = []
- os.mkdir(test_support.TESTFN)
-
-
- def tearDown(self):
- for name in self.files:
- os.unlink(name)
-
- os.rmdir(test_support.TESTFN)
-
-
- def check_tempfile(self, name):
- self.failIf(os.path.exists(name), 'file already exists for temporary file')
- open(name, 'w')
- self.files.append(name)
-
-
- def test_tempnam(self):
- if not hasattr(os, 'tempnam'):
- return None
-
- warnings.filterwarnings('ignore', 'tempnam', RuntimeWarning, 'test_os$')
- self.check_tempfile(os.tempnam())
- name = os.tempnam(test_support.TESTFN)
- self.check_tempfile(name)
- name = os.tempnam(test_support.TESTFN, 'pfx')
- self.assert_(os.path.basename(name)[:3] == 'pfx')
- self.check_tempfile(name)
-
-
- def test_tmpfile(self):
- if not hasattr(os, 'tmpfile'):
- return None
-
- fp = os.tmpfile()
- fp.write('foobar')
- fp.seek(0, 0)
- s = fp.read()
- fp.close()
- self.assert_(s == 'foobar')
-
-
- def test_tmpnam(self):
- import sys
- if not hasattr(os, 'tmpnam'):
- return None
-
- warnings.filterwarnings('ignore', 'tmpnam', RuntimeWarning, 'test_os$')
- name = os.tmpnam()
- if sys.platform in ('win32',):
- self.failIf(os.path.exists(name), 'file already exists for temporary file')
- else:
- self.check_tempfile(name)
-
-
-
- class StatAttributeTests(unittest.TestCase):
-
- def setUp(self):
- os.mkdir(test_support.TESTFN)
- self.fname = os.path.join(test_support.TESTFN, 'f1')
- f = open(self.fname, 'wb')
- f.write('ABC')
- f.close()
-
-
- def tearDown(self):
- os.unlink(self.fname)
- os.rmdir(test_support.TESTFN)
-
-
- def test_stat_attributes(self):
- if not hasattr(os, 'stat'):
- return None
-
- import stat
- result = os.stat(self.fname)
- self.assertEquals(result[stat.ST_SIZE], 3)
- self.assertEquals(result.st_size, 3)
- import sys
- members = dir(result)
- for name in dir(stat):
- if name[:3] == 'ST_':
- attr = name.lower()
- self.assertEquals(getattr(result, attr), result[getattr(stat, name)])
- self.assert_(attr in members)
- continue
-
-
- try:
- result[200]
- self.fail('No exception thrown')
- except IndexError:
- pass
-
-
- try:
- result.st_mode = 1
- self.fail('No exception thrown')
- except TypeError:
- pass
-
-
- try:
- result.st_rdev = 1
- self.fail('No exception thrown')
- except (AttributeError, TypeError):
- pass
-
-
- try:
- result.parrot = 1
- self.fail('No exception thrown')
- except AttributeError:
- pass
-
-
- try:
- result2 = os.stat_result((10,))
- self.fail('No exception thrown')
- except TypeError:
- pass
-
-
- try:
- result2 = os.stat_result((0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14))
- except TypeError:
- pass
-
-
-
- def test_statvfs_attributes(self):
- if not hasattr(os, 'statvfs'):
- return None
-
- import statvfs
-
- try:
- result = os.statvfs(self.fname)
- except OSError:
- e = None
- import errno
- if e.errno == errno.ENOSYS:
- return None
-
- except:
- e.errno == errno.ENOSYS
-
- self.assertEquals(result.f_bfree, result[statvfs.F_BFREE])
- members = dir(result)
- for name in dir(statvfs):
- if name[:2] == 'F_':
- attr = name.lower()
- self.assertEquals(getattr(result, attr), result[getattr(statvfs, name)])
- self.assert_(attr in members)
- continue
-
-
- try:
- result.f_bfree = 1
- self.fail('No exception thrown')
- except TypeError:
- pass
-
-
- try:
- result.parrot = 1
- self.fail('No exception thrown')
- except AttributeError:
- pass
-
-
- try:
- result2 = os.statvfs_result((10,))
- self.fail('No exception thrown')
- except TypeError:
- pass
-
-
- try:
- result2 = os.statvfs_result((0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14))
- except TypeError:
- pass
-
-
-
- from test import mapping_tests
-
- class EnvironTests(mapping_tests.BasicTestMappingProtocol):
- '''check that os.environ object conform to mapping protocol'''
- type2test = None
-
- def _reference(self):
- return {
- 'KEY1': 'VALUE1',
- 'KEY2': 'VALUE2',
- 'KEY3': 'VALUE3' }
-
-
- def _empty_mapping(self):
- os.environ.clear()
- return os.environ
-
-
- def setUp(self):
- self._EnvironTests__save = dict(os.environ)
- os.environ.clear()
-
-
- def tearDown(self):
- os.environ.clear()
- os.environ.update(self._EnvironTests__save)
-
-
-
- class WalkTests(unittest.TestCase):
- '''Tests for os.walk().'''
-
- def test_traversal(self):
- import os
- join = join
- import os.path
- sub1_path = join(test_support.TESTFN, 'SUB1')
- sub11_path = join(sub1_path, 'SUB11')
- sub2_path = join(test_support.TESTFN, 'SUB2')
- tmp1_path = join(test_support.TESTFN, 'tmp1')
- tmp2_path = join(sub1_path, 'tmp2')
- tmp3_path = join(sub2_path, 'tmp3')
- os.makedirs(sub11_path)
- os.makedirs(sub2_path)
- for path in (tmp1_path, tmp2_path, tmp3_path):
- f = file(path, 'w')
- f.write("I'm " + path + ' and proud of it. Blame test_os.\n')
- f.close()
-
- all = list(os.walk(test_support.TESTFN))
- self.assertEqual(len(all), 4)
- flipped = all[0][1][0] != 'SUB1'
- all[0][1].sort()
- self.assertEqual(all[0], (test_support.TESTFN, [
- 'SUB1',
- 'SUB2'], [
- 'tmp1']))
- self.assertEqual(all[1 + flipped], (sub1_path, [
- 'SUB11'], [
- 'tmp2']))
- self.assertEqual(all[2 + flipped], (sub11_path, [], []))
- self.assertEqual(all[3 - 2 * flipped], (sub2_path, [], [
- 'tmp3']))
- all = []
- for root, dirs, files in os.walk(test_support.TESTFN):
- all.append((root, dirs, files))
- if 'SUB1' in dirs:
- dirs.remove('SUB1')
- continue
-
- self.assertEqual(len(all), 2)
- self.assertEqual(all[0], (test_support.TESTFN, [
- 'SUB2'], [
- 'tmp1']))
- self.assertEqual(all[1], (sub2_path, [], [
- 'tmp3']))
- all = list(os.walk(test_support.TESTFN, topdown = False))
- self.assertEqual(len(all), 4)
- flipped = all[3][1][0] != 'SUB1'
- all[3][1].sort()
- self.assertEqual(all[3], (test_support.TESTFN, [
- 'SUB1',
- 'SUB2'], [
- 'tmp1']))
- self.assertEqual(all[flipped], (sub11_path, [], []))
- self.assertEqual(all[flipped + 1], (sub1_path, [
- 'SUB11'], [
- 'tmp2']))
- self.assertEqual(all[2 - 2 * flipped], (sub2_path, [], [
- 'tmp3']))
- for root, dirs, files in os.walk(test_support.TESTFN, topdown = False):
- for name in files:
- os.remove(join(root, name))
-
- for name in dirs:
- os.rmdir(join(root, name))
-
-
- os.rmdir(test_support.TESTFN)
-
-
-
- class MakedirTests(unittest.TestCase):
-
- def setUp(self):
- os.mkdir(test_support.TESTFN)
-
-
- def test_makedir(self):
- base = test_support.TESTFN
- path = os.path.join(base, 'dir1', 'dir2', 'dir3')
- os.makedirs(path)
- path = os.path.join(base, 'dir1', 'dir2', 'dir3', 'dir4')
- os.makedirs(path)
- self.failUnlessRaises(OSError, os.makedirs, os.curdir)
- path = os.path.join(base, 'dir1', 'dir2', 'dir3', 'dir4', 'dir5', os.curdir)
- os.makedirs(path)
- path = os.path.join(base, 'dir1', os.curdir, 'dir2', 'dir3', 'dir4', 'dir5', 'dir6')
- os.makedirs(path)
-
-
- def tearDown(self):
- path = os.path.join(test_support.TESTFN, 'dir1', 'dir2', 'dir3', 'dir4', 'dir5', 'dir6')
- while not os.path.exists(path) and path != test_support.TESTFN:
- path = os.path.dirname(path)
- os.removedirs(path)
-
-
-
- class DevNullTests(unittest.TestCase):
-
- def test_devnull(self):
- f = file(os.devnull, 'w')
- f.write('hello')
- f.close()
- f = file(os.devnull, 'r')
- self.assertEqual(f.read(), '')
- f.close()
-
-
-
- class URandomTests(unittest.TestCase):
-
- def test_urandom(self):
-
- try:
- self.assertEqual(len(os.urandom(1)), 1)
- self.assertEqual(len(os.urandom(10)), 10)
- self.assertEqual(len(os.urandom(100)), 100)
- self.assertEqual(len(os.urandom(1000)), 1000)
- except NotImplementedError:
- pass
-
-
-
-
- def test_main():
- test_support.run_unittest(TemporaryFileTests, StatAttributeTests, EnvironTests, WalkTests, MakedirTests, DevNullTests, URandomTests)
-
- if __name__ == '__main__':
- test_main()
-
-